Cortex: Consolidating Context into Parametric Memory
Cortex:将上下文收益固化进 LLM 参数记忆
🧭 LLM 记忆固化 · 参数更新 ★ 深度解读
Zhejiang University · ByteDance Seed · Tsinghua University
用前瞻性重演与 FFN-only 低秩/结构化更新,把上下文增益蒸馏为可堆叠的参数记忆,兼顾 CR 与 RR
📌 工作参考价值(4 点)
- Agent 经验与临时规则沉淀:可把商品知识、活动政策、API 文档、成功轨迹等上下文,先用冻结模型生成“预期查询+上下文条件回答”,再蒸馏到 0.74M–2.73M 的小型 FFN adapter;无需把完整上下文塞进 prompt 或检索库,适合高频、确定性的商户规则/接口知识固化。
- 架构选择:只更新 FFN/MoE,冻结 attention、router 和 backbone;Structural 新增非线性通道比低秩 Synaptic 更擅长代码/API 类可复用过程,且 MoE 用 shared branch 给所有 token 直接训练曝光,减少遗忘、提升 CR。
- 训练目标 trick:用 top-K_v 截断 KL + 对齐后 FFN hidden-state MSE 监督,teacher 带上下文、student 不带;这对少样本、要保 base 能力的高保续更新很有效。历史 query replay 只留 query、不存 context,降低隐私和存储成本。
- 落地边界:CR/RR 的权衡在实体邻近知识上仍明显,组合推理/可移植性(portability)是短板;建议只固化高频、低歧义技能,开放生成仍保留 RAG/prompt 兜底。